#!/bin/bash

nnUNet_dataset=/home/yusongli/Templates/EsoSeg

export nnUNet_n_proc_DA=12
export nnUNet_raw="${nnUNet_dataset}/nnUNet_raw"
export nnUNet_preprocessed="${nnUNet_dataset}/nnUNet_preprocessed"
export nnUNet_results="${nnUNet_dataset}/nnUNet_results"
export CUDA_DEVICE_ORDER=PCI_BUS_ID
export CUDA_VISIBLE_DEVICES=0


function preprocess () {
    nnUNetv2_plan_and_preprocess -d 000
}


function train () {
    nnUNetv2_train \
        000 \
        3d_fullres \
        0
}


train
